@echo off
REM DiffPool 依赖安装脚本（Windows批处理）
REM 解决 setuptools 和版本兼容性问题

echo ========================================
echo DiffPool 依赖包安装脚本
echo ========================================
echo.

echo [1/6] 升级 pip 和 setuptools...
python -m pip install --upgrade pip
pip install --upgrade setuptools wheel
echo.

echo [2/6] 安装核心科学计算库...
pip install numpy==1.24.3
pip install scipy==1.11.4
echo.

echo [3/6] 安装 PyTorch...
pip install torch==2.2.0 torchvision==0.17.0
echo.

echo [4/6] 安装机器学习工具...
pip install scikit-learn==1.3.2
echo.

echo [5/6] 安装图处理库...
pip install networkx==3.1
pip install python-louvain==0.16
echo.

echo [6/6] 安装可视化和日志工具...
pip install matplotlib==3.8.2
pip install tensorboard==2.15.1
pip install tensorboardX==2.6.2.2
echo.

echo ========================================
echo 安装完成！
echo ========================================
echo.

echo 验证安装...
python -c "import torch; import numpy; import networkx; import sklearn; print('✓ 所有依赖包安装成功！')"
echo.

pause

